import torch
import matplotlib.pyplot as plt

# 散度
noise = 0.2
# 簇类A
a = torch.normal(0.5, noise, (20, 2))
plt.plot(a[:, 0], a[:, 1], 'ro')

# 簇类B
b = torch.normal(1.5, noise, (20, 2))
plt.plot(b[:, 0], b[:, 1], 'bo')

plt.show()
